/*
 * @Descripttion: 
 * @version: 
 * @Author: lily
 * @Date: 2021-05-11 09:43:50
 * @LastEditors: lily
 * @LastEditTime: 2021-05-11 14:28:57
 */
/*
 * @lc app=leetcode.cn id=75 lang=javascript
 *
 * [75] 颜色分类
 */

// @lc code=start
/**
 * @param {number[]} nums
 * @return {void} Do not return anything, modify nums in-place instead.
 */
//  思想：双指针
//  0 移到最前面，2 移到最后面
//  2 如果换出来的不是 1，那么指针要回退，因为 0 和 2 都是需要再次移动的

var sortColors = function (nums) {
    // p指向要换的头 q指向要换的尾
    let p = 0, q = nums.length - 1
    for (let i = 0; i <= q; i++) {
        if (nums[i] === 0) {
            [nums[i], nums[p]] = [nums[p], nums[i]]
            p++
        }
        if (nums[i] === 2) {
            [nums[i], nums[q]] = [nums[q], nums[i]]
            q--
            // 回退
            if (nums[i] !== 1) {
                i--
            }
        }
    }
};
// @lc code=end
let arr = [2, 2, 1, 0, 2, 1, 0, 1, 2, 1, 0]
sortColors(arr)
console.log(arr);